/*
** Copyright (C) 2010 Calin Radoni
**
** This file is part of CHScanner.
**
** CHScanner is free software: you can redistribute it and/or modify
** it under the terms of the GNU General Public License as published by
** the Free Software Foundation, either version 3 of the License, or
** (at your option) any later version.
**
** CHScanner is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
** GNU General Public License for more details.
**
** You should have received a copy of the GNU General Public License
** along with CHScanner. If not, see <http://www.gnu.org/licenses/>.
*/

#pragma once
#include "scanmethod.h"

class ScanMethodShutdown : public ScanMethod
{
protected:
	virtual int InternalInit(void);
	virtual int InternalExecute(void);

	virtual void ComputeRandomValues(void);

	TCHAR userName[CREDUI_MAX_USERNAME_LENGTH + 1];
	TCHAR userPass[CREDUI_MAX_PASSWORD_LENGTH + 1];
	bool interactiveRun;

	bool    xReboot;
	bool    xForced;
	int     xTimeOut;
	CString xMessage;
	bool ShutItDown(CString);

	HANDLE           xUserHandle;
	HANDLE           xThreadHandle;
	HANDLE           xTokenHandle;
	TOKEN_PRIVILEGES xPrivileges;
	bool             xImpersonated;
	bool ElevatePrivileges(void);
	void RestorePrivileges(void);

public:
	ScanMethodShutdown(void);
	~ScanMethodShutdown(void);

	void CleanUpUserCredentials(void);
	void SetUserCredentials(TCHAR*, TCHAR*, bool);
};
